package com.lyy.bigevent.utils;

import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;

/**
 * MD5加密工具类
 * 用于对密码进行加密处理
 */
public class Md5Util {

    /**
     * 对字符串进行MD5加密
     *
     * @param input 原始字符串
     * @return 加密后的32位小写MD5字符串
     */
    public static String md5(String input) {
        if (input == null || input.isEmpty()) {
            return null;
        }

        try {
            // 创建MessageDigest实例，指定MD5算法
            MessageDigest md = MessageDigest.getInstance("MD5");

            // 计算MD5哈希值
            byte[] hash = md.digest(input.getBytes());

            // 将字节数组转换为十六进制字符串
            return bytesToHex(hash);

        } catch (NoSuchAlgorithmException e) {
            throw new RuntimeException("MD5算法不可用", e);
        }
    }

    /**
     * 对密码进行MD5加密（带盐值版本）
     * 推荐使用此方法增强安全性
     *
     * @param password 原始密码
     * @param salt     盐值
     * @return 加密后的MD5字符串
     */
    public static String md5WithSalt(String password, String salt) {
        if (password == null || salt == null) {
            return null;
        }
        // 将密码和盐值组合后再进行MD5加密
        return md5(password + salt);
    }

    /**
     * 验证密码是否正确
     *
     * @param inputPassword 用户输入的密码
     * @param storedHash    数据库中存储的加密密码
     * @return 是否匹配
     */
    public static boolean verifyPassword(String inputPassword, String storedHash) {
        if (inputPassword == null || storedHash == null) {
            return false;
        }
        String inputHash = md5(inputPassword);
        return inputHash.equals(storedHash);
    }

    /**
     * 验证带盐值的密码
     *
     * @param inputPassword 用户输入的密码
     * @param storedHash    数据库中存储的加密密码
     * @param salt          盐值
     * @return 是否匹配
     */
    public static boolean verifyPasswordWithSalt(String inputPassword, String storedHash, String salt) {
        if (inputPassword == null || storedHash == null || salt == null) {
            return false;
        }
        String inputHash = md5WithSalt(inputPassword, salt);
        return inputHash.equals(storedHash);
    }

    /**
     * 字节数组转换为十六进制字符串
     *
     * @param bytes 字节数组
     * @return 十六进制字符串
     */
    private static String bytesToHex(byte[] bytes) {
        StringBuilder hexString = new StringBuilder();
        for (byte b : bytes) {
            String hex = Integer.toHexString(0xff & b);
            if (hex.length() == 1) {
                hexString.append('0');
            }
            hexString.append(hex);
        }
        return hexString.toString();
    }

    /**
     * 生成随机盐值
     *
     * @param length 盐值长度
     * @return 随机盐值
     */
    public static String generateSalt(int length) {
        if (length <= 0) {
            length = 16; // 默认长度
        }

        String characters = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789!@#$%^&*()";
        StringBuilder salt = new StringBuilder();

        for (int i = 0; i < length; i++) {
            int index = (int) (Math.random() * characters.length());
            salt.append(characters.charAt(index));
        }

        return salt.toString();
    }

    /**
     * 生成默认长度的随机盐值（16位）
     */
    public static String generateSalt() {
        return generateSalt(16);
    }
}
